Skip to content

Piyushi-0/SEA-NN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

53 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Improving Attribution Methods by Learning Submodular Functions

Accepted at the 25th International Conference on Artificial Intelligence and Statistics (AISTATS '22).

Code in PyTorch | Talk | Slides | Link to arxiv pre-print | Link to the version accepted at ICML '20 workshops (XXAI, WHI)

This work explores the novel idea of learning a submodular scoring function to improve the specificity of existing feature attribution methods. A novel formulation for learning a deep submodular set function that is consistent with the real-valued attribution maps obtained by existing attribution methods is proposed. The final attribution value of a feature is then defined as the marginal gain in the induced submodular score of the feature in the context of other highly attributed features, thus decreasing the attribution of redundant yet discriminatory features. Experiments on multiple datasets illustrate that the proposed attribution method achieves higher specificity along with good discriminative power.

Acknowledgements

The first author is especially grateful to the mentorship of Dr Bilal Alsallakh and for being supported by the Google PhD Fellowship.


The first author's research interests have changed. Please contact https://github.com/peppermenta for discussions.

About

Code for our AISTATS '22 paper: Improving Attribution Methods by Learning Submodular Functions.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published